package com.cyzy.controller;
import com.cyzy.config.MinioConfig;
import com.cyzy.utils.MinioUtils;
import jakarta.servlet.http.HttpServletResponse;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.compress.utils.IOUtils;
import org.apache.commons.lang3.StringUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;

import java.io.InputStream;
import java.util.HashMap;
import java.util.Map;

/**
 * 对象存储服务控制器
 * 提供文件上传、下载、删除等操作的接口
 * @Slf4j 自动为类创建一个名为 log 的 SLF4J Logger 实例
 * @author LIGHT
 */
@Slf4j
@RestController
@RequestMapping("/oss")
public class OSSController {

    /**
     * MinIO配置类，包含存储桶名称等配置信息
     */
    @Autowired
    private MinioConfig minioConfig;
    @Autowired
    private MinioUtils minioUtils;
    /**
     * 文件上传接口
     * @param file 上传的文件
     * @return 上传结果信息，包含文件名
     */
    @PostMapping("/upload")
    public Map<String, Object> upload(@RequestParam("file") MultipartFile file){
        Map<String, Object> result = new HashMap<>();
        try {
            // 获取原始文件名
            String fileName = file.getOriginalFilename();
            System.out.println("原始文件名:"+fileName);
            // 生成新的文件名：时间戳 + 原文件扩展名
            String newFileName = System.currentTimeMillis() +"."+ StringUtils.substringAfterLast(fileName,".");
            System.out.println(newFileName);
//            // 获取文件类型
            String contentType = file.getContentType();
            // 上传文件到MinIO
            minioUtils.uploadFile(minioConfig.getBucketName(), file, newFileName, contentType);
            
            // 获取文件访问URL
            String fileUrl = minioUtils.getPresignedObjectUrl(minioConfig.getBucketName(), newFileName);
            
            result.put("success", true);
            result.put("message", "上传成功");
            result.put("fileName", newFileName);
            result.put("fileUrl", fileUrl);
            return result;
        } catch (Exception e) {
            log.error("上传失败", e);
            result.put("success", false);
            result.put("message", "上传失败: " + e.getMessage());
            return result;
        }
    }

    /**
     * 文件删除接口
     * @param fileName 要删除的文件名
     */
    @DeleteMapping("/delete")
    public void delete(@RequestParam("fileName") String fileName){
        minioUtils.removeFile(minioConfig.getBucketName(), fileName);
    }

    /**
     * 获取文件状态信息接口
     * @param fileName 文件名
     * @return 文件状态信息
     */
    @GetMapping("/info")
    public String getFileStatusInfo(@RequestParam("fileName") String fileName){
        return minioUtils.getFileStatusInfo(minioConfig.getBucketName(), fileName);
    }

    /**
     * 获取文件访问URL接口
     * @param fileName 文件名
     * @return 预签名的文件访问URL
     */
    @GetMapping("/url")
    public String getPresignedObjectUrl(@RequestParam("fileName") String fileName){
        return minioUtils.getPresignedObjectUrl(minioConfig.getBucketName(), fileName);
    }

    /**
     * 文件下载接口
     * @param fileName 要下载的文件名
     * @param response HTTP响应对象
     */
    @GetMapping("/download")
    public void download(@RequestParam("fileName") String fileName, HttpServletResponse response){
        try {
            // 从MinIO获取文件输入流
            InputStream fileInputStream = minioUtils.getObject(minioConfig.getBucketName(), fileName);
            // 设置响应头，指定文件下载方式
            response.setHeader("Content-Disposition", "attachment;filename=" + fileName);
            response.setContentType("application/force-download");
            response.setCharacterEncoding("UTF-8");
            // 将文件流复制到响应输出流
            IOUtils.copy(fileInputStream, response.getOutputStream());
        } catch (Exception e) {
            log.error("下载失败");
        }
    }
}
